package com.kuaiyin.interceptor;

import com.kuaiyin.exception.BaseException;
import com.kuaiyin.utils.JwtUtils;
import io.jsonwebtoken.Claims;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;

import java.util.Date;

/**
 * 登录校验拦截器
 */
@Slf4j
@Component
public class LoginCheckInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 1. 获取请求url。
        String uri = request.getRequestURI();
        log.info("uri = {}", uri);

        // 2. 判断请求url中是否包含login，如果包含，说明是登录操作，放行。
        if (uri.contains("login")   || uri.contains("captcha")) {
            return true;
        }

        // 3. 获取请求头中的令牌（token）。
        String token = request.getHeader("token");
        request.setAttribute("CURRENT_USER_token", token);  // 将token放入请求域中，用于后续的拦截器获取
        // 4. 判断令牌是否存在，如果不存在，响应401。
        if (token == null || token.isEmpty()) {
            throw BaseException.of(HttpStatus.UNAUTHORIZED.value(), "令牌不存在！");
        }

        // 5. 验证令牌是否有效（包括过期检查）。
        try {
            Claims claims = JwtUtils.parseJWT(token);
            Date expirationDate = claims.getExpiration();

            if (expirationDate.before(new Date())) {
                throw BaseException.of(HttpStatus.UNAUTHORIZED.value(), "令牌已过期！");
            }

        } catch (Exception e) {
            throw BaseException.of(HttpStatus.UNAUTHORIZED.value(), "令牌无效或解析失败");
        }

        // 6. 放行。
        return true;
    }
}
